Input tensor is passed to custom layer with wrong shape